from torchvision.models.segmentation import deeplabv3_resnet101
model = deeplabv3_resnet101(num_classes=10,)